#ifndef AVALON_UIDSERVICETESTS_HPP
#define AVALON_UIDSERVICETESTS_HPP
#include "Avalon/IOTests/IOTests.hpp"
#include "Avalon/ServiceLocator/ServiceLocator.hpp"
#include "Avalon/UidService/UidService.hpp"

namespace Avalon {
namespace UidService {
namespace Tests {
  class MockUidDataStore;

  //! Registers a UidService on a mock server connection.
  /*!
    \param serviceClient The ServiceClient to register the service on.
    \param serverConnection The connection to the mock UID server.
    \param dataShuttle The name of the DataShuttle to use.
  */
  void RegisterTestUidService(
    ServiceLocator::ServiceLocatorClient* serviceClient,
    IO::Tests::MockServerConnection* serverConnection,
    const std::string& dataShuttle);
}
}
}

#endif // AVALON_UIDSERVICETESTS_HPP
